#切换不同环境
from configparser import ConfigParser

class Config:
    def __init__(self):
        self.config = ConfigParser()
        self.config.read(os.path.join(os.path.dirname(__file__), 'config.ini'))
        self.base_url = self.config.get('ENV', 'base_url')
        self.db_host = self.config.get('DATABASE', 'db_host')
        # 其他配置项...

    def switch_env(self, env):
        """切换环境，更新配置"""
        self.config.set('ENV', 'env', env)
        if env == 'test':
            self.config.set('ENV', 'base_url', 'http://test-api.example.com')
            self.config.set('DATABASE', 'db_host', '127.0.0.1')
        elif env == 'prod':
            self.config.set('ENV', 'base_url', 'http://prod-api.example.com')
            self.config.set('DATABASE', 'db_host', 'prod-db.example.com')
        # 写入修改（可选，若要持久化修改配置文件）
        with open(os.path.join(os.path.dirname(__file__), 'config.ini'), 'w') as f:
            self.config.write(f)